import uiautomator2 as u2
from zz import zz_code


# 快速保存手机截图

def main():
    d = u2.connect(zz_code.env('DEVICE_ID'))
    img = d.screenshot()
    img.save(zz_code.env('DEVICE_SAVE_PATH', 'D:/_download/phone.png'))


if __name__ == '__main__':
    main()
